import sys
import jieba
from wordcloud import WordCloud  # 生成词云
import matplotlib.pyplot as plt  # 创建图片

text = sys.argv[1]  # 获取node传过来的参数

wordText = jieba.cut(text)  # 分词
wordText = " ".join(wordText)  # 使用空格作分隔符，转成字符串

# 这个东西不支持中文，因此需要中文字体支持
wordCloud = WordCloud(
    font_path=r"C:\Users\Administrator\Documents\personal\chabaidao\server\app\script\puppet\YouSheBiaoTiHei.ttf",  # 设置字体
    background_color="white",  # 设置背景颜色
    max_words=200,  # 设置最大显示的字数
    max_font_size=100,  # 设置字体最大值
    width=500,  # 设置图片的宽度
    height=500
).generate(wordText) #生成词云
print(wordCloud)

plt.imshow(wordCloud)  # 创建图像
plt.axis("off")  # 关闭坐标轴
plt.show()  # 显示图片
